# Filename: backup_ver4.py
import os
import time

# 1.需要备份的文件目录
source = ['F:\\python\\code']

# 2. 备份文件到目标路径
target_dir = 'F:\\python\\backup\\'

# 3. The files are backed up into a zip file.
# 4. The current day is the name of the subdirectory in the main directory
today = target_dir + time.strftime('%Y%m%d')
#bkname = ['code']
# The current time is the name of the zip archive
now = time.strftime('%H%M%S')

# Take a comment from the user to create the name of the zip file
comment = input('Enter a comment -->')
if len(comment)==0: 
    target = today+os.sep+now+'.zip'
else:
    target = today+os.sep+now+'_'+\
             comment.replace(' ','_')+'.zip'
    # Notice the backslash!

# Create the subdirectory if it isn't already there
if not os.path.exists(today):
    os.mkdir(today)  # make directory
    print('Successfully created directory', today)

# 5. 用winrar的rar命令压缩文件，但首先要安装有winrar且设置winrar到环境变量的路径path中
zip_command = "rar a %s %s" %(target,''.join(source))

# Run the backup
#设置winrar到path环境中，这里已经手动添加，如图
#os.system('set Path=%Path%;C:\Program Files\WinRAR')
if os.system(zip_command)==0:
    print('Successful backup to', target)
else:
    print('Backup FAILED')
